from time import sleep

from selenium.webdriver.common.by import By

from workchat.basetool.basetool import BaseTool
from workchat.workchat_po.contact_page import ContactPage


class MainPage(BaseTool):
    def click_contact(self):
        sleep(2)
        self.driver.find_element(By.ID, "menu_contacts").click()
        return ContactPage()
